tensorflow 2.0 fine tuning